// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

use horaedbproto::storage::{RouteRequest as RouteRequestPb, RouteResponse};
use router::RouteRequest;

use crate::{error, metrics::GRPC_HANDLER_COUNTER_VEC, Context, Proxy};

impl Proxy {
    pub async fn handle_route(&self, _ctx: Context, req: RouteRequestPb) -> RouteResponse {
        let request = RouteRequest::new(req, true);
        let routes = self.route(request).await;

        let mut resp = RouteResponse::default();
        match routes {
            Err(e) => {
                GRPC_HANDLER_COUNTER_VEC.route_failed.inc();

                error!("Failed to handle route, err:{e}");
                resp.header = Some(error::build_err_header(e));
            }
            Ok(v) => {
                GRPC_HANDLER_COUNTER_VEC.route_succeeded.inc();

                resp.header = Some(error::build_ok_header());
                resp.routes = v;
            }
        }
        resp
    }
}
